package com.seu.me.lock;

import redis.clients.jedis.Jedis;

import java.util.Collections;

/**
 * Redis分布式锁：互斥性、同一性、可重入性
 * 推荐写法
 * SET key value [EX seconds] [PX milliseconds] [NX|XX]
 * EX second ：设置键的过期时间为 second 秒。
 * PX millisecond ：设置键的过期时间为 millisecond 毫秒。
 * NX ：只在键不存在时，才对键进行设置操作。
 * XX ：只在键已经存在时，才对键进行设置操作。
 *
 * @author liangfeihu
 * @since 2019/6/11
 */
public class RedisLockV2 {
    private static final String LOCK_SUCCESS = "OK";
    private static final String SET_IF_NOT_EXIST = "NX";
    private static final String SET_WITH_EXPIRE_TIME = "EX";

    private static final Long RELEASE_SUCCESS = 1L;

    /**
     * 使用redis的set命令实现获取分布式锁
     * 命令 SET resource-name-key anystringVal NX EX max-lock-time 是一种在 Redis 中实现锁的简单方法。
     * 客户端执行以上的命令：
     * 如果服务器返回 OK ，那么这个客户端获得锁。
     * 如果服务器返回 NIL ，那么客户端获取锁失败，可以在稍后再重试。
     * 设置的过期时间到达之后，锁将自动释放。
     *
     * @param lockKey    可以就是锁
     * @param requestId  请求ID，保证同一性
     * @param expireTime 过期时间，避免死锁(单位秒)
     * @return
     */
    public static boolean getLock(Jedis jedis, String lockKey, String requestId, int expireTime) {
        //NX:保证互斥性
        String result = jedis.set(lockKey, requestId, SET_IF_NOT_EXIST, SET_WITH_EXPIRE_TIME, expireTime);
        if (LOCK_SUCCESS.equals(result)) {
            return true;
        }
        return false;
    }


    /**
     * 释放分布式锁
     * 不使用固定的字符串作为键的值，而是设置一个不可猜测（non-guessable）的长随机字符串，作为口令串（token）。
     * 不使用 DEL 命令来释放锁，而是发送一个 Lua 脚本，这个脚本只在客户端传入的值和键的口令串相匹配时，才对键进行删除。
     *
     * @param lockKey
     * @param requestId
     */
    public static boolean releaseLock(Jedis jedis, String lockKey, String requestId) {
        // 可以防止持有过期锁的客户端误删现有锁的情况出现。
        String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
        Object result = jedis.eval(script, Collections.singletonList(lockKey), Collections.singletonList(requestId));

        if (result.equals(RELEASE_SUCCESS)) {
            return true;
        }
        return false;
    }

}
